//
//  Problem13.swift
//  LeetCode - 罗马数字转整数
//
//  Created by 武侠 on 2020/7/3.
//  Copyright © 2020 zhulongiMac02. All rights reserved.
//

import UIKit
/*
罗马数字转整数
罗马数字包含以下七种字符: I， V， X， L，C，D 和 M。
字符          数值
I             1
V             5
X             10
L             50
C             100
D             500
M             1000
例如:
    2:  写做 II ，即为两个并列的 1。
    12: 写做 XII ，即为 X + II 。
    27: 写做 XXVII, 即为 XX + V + II 。

通常情况下，罗马数字中小的数字在大的数字的右边。但也存在特例，例如:
    4:  不写做 IIII，而是 IV。数字 1 在数字 5 的左边，所表示的数等于大数 5 减小数 1 得到的数值 4 。
    9:  表示为 IX。
 
这个特殊的规则只适用于以下六种情况：
I 可以放在 V (5) 和 X (10) 的左边，来表示 4 和 9。
X 可以放在 L (50) 和 C (100) 的左边，来表示 40 和 90。
C 可以放在 D (500) 和 M (1000) 的左边，来表示 400 和 900。
 
给定一个罗马数字，将其转换成整数。输入确保在 1 到 3999 的范围内。
示例 1:   输入: "III"       输出: 3
示例 2:   输入: "IV"        输出: 4
示例 3:   输入: "IX"        输出: 9
示例 4:   输入: "LVIII"     输出: 58
解释: L = 50, V= 5, III = 3.
示例 5:   输入: "MCMXCIV"   输出: 1994
解释: M = 1000, CM = 900, XC = 90, IV = 4.
*/
class Problem13: NSObject {
    func solution() {
        let nums = ["IV", "III", "IX", "LVIII", "MCMXCIV"]
        for num in nums {
            print(num, " = ", romanToInt(num))
        }
    }
    func romanToInt(_ s: String) -> Int {
        var startIndex = s.startIndex
        var rs: Int = 0
        while startIndex != s.endIndex {
            var nextIndex = s.index(after: startIndex)
            let str = s[startIndex..<nextIndex]
            if str == "M" {
                rs += 1000
            } else if str == "D" {
                rs += 500
            } else if str == "C" {
                // C 可以放在 D (500) 和 M (1000) 的左边，来表示 400 和 900。
                if nextIndex != s.endIndex {
                    let nnextIndex = s.index(after: nextIndex)
                    let nextStr = s[nextIndex..<nnextIndex]
                    if nextStr == "D" {
                        rs += 400
                        nextIndex = nnextIndex
                    } else if nextStr == "M" {
                        rs += 900
                        nextIndex = nnextIndex
                    } else {
                        rs += 100
                    }
                } else {
                    rs += 100
                }
            } else if str == "L" {
                rs += 50
            } else if str == "X" {
                // X 可以放在 L (50) 和 C (100) 的左边，来表示 40 和 90。
                if nextIndex != s.endIndex {
                    let nnextIndex = s.index(after: nextIndex)
                    let nextStr = s[nextIndex..<nnextIndex]
                    if nextStr == "L" {
                        rs += 40
                        nextIndex = nnextIndex
                    } else if nextStr == "C" {
                        rs += 90
                        nextIndex = nnextIndex
                    } else {
                        rs += 10
                    }
                } else {
                    rs += 10
                }
            } else if str == "V" {
                rs += 5
            } else if str == "I" {
                // I 可以放在 V (5) 和 X (10) 的左边，来表示 4 和 9。
                if nextIndex != s.endIndex {
                    let nnextIndex = s.index(after: nextIndex)
                    let nextStr = s[nextIndex..<nnextIndex]
                    if nextStr == "V" {
                        rs += 4
                        nextIndex = nnextIndex
                    } else if nextStr == "X" {
                        rs += 9
                        nextIndex = nnextIndex
                    } else {
                        rs += 1
                    }
                } else {
                    rs += 1
                }
            }
            startIndex = nextIndex
        }
        return rs
    }
}
